import os
import numpy as np
import sys
import torch

if __name__ == '__main__':
    a = torch.Tensor([[1,3,2.],[2.3,4.4,2]])

    b = [a, a, a]
    b = torch.stack(b, dim=0)
    print(b)